package demo4.mapper;

import demo4.pojo.User;
import demo4.util.IDBUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.sql.*;
import java.util.ArrayList;
import java.util.List;

@Component
public class Usermapper implements IUserMapper{
    @Autowired
    private IDBUtil dbUtil;
    @Override
    public int insert(User user){
        System.out.println("UserMapper:向user表中插入用户数据...");
        try (Connection connection = dbUtil.getConnection()) {
            String sql = "INSERT INTO user(username,password,nickname,age )" +
                         "VALUES(?,?,?,?)";
            PreparedStatement ps = connection.prepareStatement(sql);
            ps.setString(1,user.getUsername());
            ps.setString(2,user.getPassword());
            ps.setString(3,user.getNickname());
            ps.setInt(4,user.getAge());
            return ps.executeUpdate();
        } catch (SQLException e) {
            e.printStackTrace();
        }
        System.out.println("UserMapper:执行Insert操作...");
        System.out.println("UserMapper:执行完毕，返回影响的条目数");
        return 0;
    }

    @Override
    public User selectByUsername(String username){
            try (Connection connection = dbUtil.getConnection()) {
                String sql = "SELECT username,password,nickname,age " +
                             "FROM user " +
                             "WHERE username = ?";
                PreparedStatement ps = connection.prepareStatement(sql);
                ps.setString(1,username);
                ResultSet rs = ps.executeQuery();
                if(rs.next()){
                    User user = new User();
                    user.setUsername(rs.getString("username"));
                    user.setPassword(rs.getString("password"));
                    user.setNickname(rs.getString("nickname"));
                    user.setAge(rs.getInt("age"));
                    return user;
                }
            } catch (SQLException e) {
                e.printStackTrace();
            }
        return null;
    }

    @Override
    public List<User> selectAll(){
        List<User> users = new ArrayList<>();
        try (Connection connection = dbUtil.getConnection()) {
            String sql = "SELECT username,nickname,age " +
                         "FROM user";
            Statement st = connection.createStatement();
            ResultSet rs = st.executeQuery(sql);
            while(rs.next()){
                User user = new User();
                user.setUsername(rs.getString("username"));
                user.setNickname(rs.getString("nickname"));
                user.setAge(rs.getInt("age"));
                users.add(user);
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return users;
    }

    @Override
    public int update(User user){
        try (Connection connection = dbUtil.getConnection()) {
            String sql = "UPDATE user " +
                         "SET password = ?,nickname = ?,age = ? " +
                         "WHERE username = ?";
            PreparedStatement ps = connection.prepareStatement(sql);
            ps.setString(1,user.getPassword());
            ps.setString(2,user.getNickname());
            ps.setInt(3,user.getAge());
            ps.setString(4,user.getUsername());
            int num = ps.executeUpdate();
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return 0;
    }
}
